package com.brainhealth.common.enums;

import java.util.ArrayList;
import java.util.List;

/**
 * AI回话角色枚举
 * @author wll
 */
public enum AIRoleEnum {
    COMPLETION(0, "答案"), PROMPT(1, "问题");
    private final int code;
    private final String info;

    AIRoleEnum(int code, String info) {
        this.code = code;
        this.info = info;
    }

    public int getCode()
    {
        return code;
    }

    public String getInfo()
    {
        return info;
    }

    /**
     * 获取报告编码列表
     *
     * @return
     */
    public static List<Integer> getRoleCodeList() {
        List<Integer> list = new ArrayList<>(3);
        for (AIRoleEnum aiRoleEnum : AIRoleEnum.values()) {
            list.add(aiRoleEnum.code);
        }
        return list;
    }
}
